package com.test.lyt.uils;

import java.util.Arrays;
import java.util.List;

public class ImageUtil {

    private static final List<String> IMAGE_EXTENSIONS = Arrays.asList(
            ".jpg", ".jpeg", ".png", ".gif", ".bmp", ".tiff", ".webp"
    );

    /**
     * 判断文件名是否是图片格式
     *
     * @param fileName 文件名
     * @return 如果是图片格式则返回true，否则返回false
     */
    public static boolean isImage(String fileName) {
        if (fileName == null || fileName.isEmpty()) {
            return false;
        }

        // 获取文件扩展名
        String extension = getExtension(fileName).toLowerCase();

        // 检查扩展名是否在图片扩展名列表中
        return IMAGE_EXTENSIONS.contains(extension);
    }

    /**
     * 获取文件扩展名
     *
     * @param fileName 文件名
     * @return 文件扩展名，如果没有扩展名则返回空字符串
     */
    private static String getExtension(String fileName) {
        int lastDotIndex = fileName.lastIndexOf('.');
        if (lastDotIndex == -1) { // 文件名中没有点号
            return "";
        }
        return fileName.substring(lastDotIndex);
    }
}